package com.ushine.web.security.jwt;

import com.ushine.web.component.exception.LoginErrorEnum;
import com.ushine.web.component.exception.ServiceException;
import com.ushine.web.security.utils.RsaKeyConfig;
import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import lombok.extern.slf4j.Slf4j;
import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.util.*;

/**
 * @Author: yls
 * @Date: 2021/8/19 10:48
 * @Description:
 * @Version 1.0
 */
@Slf4j
@Component
public class JwtTokenUtil {

    @Resource
    private RsaKeyConfig rsaKeyConfig;

    @Resource
    JwtProperties jwtProperties;

    /**
     * 功能描述:
     * 〈 生成【token】令牌 〉
     *
     * @param userDetails 用户信息
     * @return : java.lang.String 令牌
     * @author : yls
     * @date : 2021/8/19 10:52
     */
    public String generateToken(UserDetails userDetails) {
        HashMap<String, Object> claims = new HashMap<>(1);
        claims.put("sub", userDetails.getUsername());
        return generateToken(claims);
    }

    /**
     * 功能描述:
     * 〈 生成令牌 〉
     *
     * @param claims 数据声明
     * @return : java.lang.String 令牌
     * @author : yls
     * @date : 2021/8/19 10:58
     */
    private String generateToken(Map<String, Object> claims) {
        Date date = new Date(System.currentTimeMillis());
        return Jwts.builder().setClaims(claims)
                .setId(createJti())
                .setIssuedAt(date)
                .setExpiration(new Date(date.getTime() + jwtProperties.getExpiration()))
                .signWith(SignatureAlgorithm.RS256, rsaKeyConfig.getPrivateKey())
                .compact();
    }

    /**
     * 功能描述:
     * 〈 从令牌中获取用户名 〉
     *
     * @param token 令牌
     * @return : java.lang.String 用户名
     * @author : yls
     * @date : 2021/8/19 11:05
     */
    public String getUsernameFromToken(String token) {
        String username;
        try {
            Claims claims = getClaimsFromToken(token);
            username = claims.getSubject();
        } catch (Exception e) {
            username = null;
        }
        return username;
    }


    /**
     * 功能描述:
     * 〈 获取数据声明 〉
     *
     * @param token 令牌
     * @return : io.jsonwebtoken.Claims
     * @author : yls
     * @date : 2021/8/19 11:01
     */
    public Claims getClaimsFromToken(String token) {
        Claims claims;
        try {
            claims = Jwts.parser().setSigningKey(rsaKeyConfig.getPublicKey()).parseClaimsJws(token).getBody();
        } catch (Exception e) {
            log.info(this.getClass() + "-getClaimsFromToken-exception：" + e);
            claims = null;
        }
        return claims;
    }

    /**
     * 功能描述:
     * 〈 生成一个【id】 〉
     *
     * @param
     * @return : java.lang.String
     * @author : yls
     * @date : 2021/8/19 10:57
     */
    private static String createJti() {
        return new String(Base64.getEncoder().encode(UUID.randomUUID().toString().getBytes()));
    }

    /**
     * 功能描述:
     * 〈 判断令牌是否过期 〉
     *
     * @param token 令牌
     * @return : java.lang.Boolean 是否过期
     * @author : yls
     * @date : 2021/8/19 11:23
     */
    public Boolean isTokenExpired(String token) {
        try {
            Claims claims = getClaimsFromToken(token);
            Date expiration = claims.getExpiration();
            return expiration.before(new Date());
        } catch (Exception e) {
            return false;
        }
    }


    public void logOutToken(String token) {
        Claims claims = getClaimsFromToken(token);
        claims.setExpiration(new Date());
    }

    /**
     * 功能描述:
     * 〈 获取过期时间 〉
     *
     * @param token token
     * @return : java.util.Date
     * @author : yls
     * @date : 2022/2/14 17:25
     */
    public Date getExpiration(String token) throws ServiceException {
        Claims claims = getClaimsFromToken(token);
        if (claims == null) {
           throw new ServiceException(LoginErrorEnum.USER_ACCOUNT_EXPIRED);
        }
        return claims.getExpiration();
    }

    /**
     * 功能描述:
     * 〈 刷新token 〉
     *
     * @param token 原令牌
     * @return : java.lang.String 新令牌
     * @author : yls
     * @date : 2021/8/19 11:28
     */
    public String refreshToken(String token) {
        String refreshedToken;
        try {
            Claims claims = getClaimsFromToken(token);
            refreshedToken = generateToken(claims);
        } catch (Exception e) {
            refreshedToken = null;
        }
        return refreshedToken;
    }

    /**
     * 功能描述:
     * 〈 验证令牌是否又有效 〉
     *
     * @param token       令牌
     * @param userDetails 用户
     * @return : java.lang.Boolean 是否有效
     * @author : yls
     * @date : 2021/8/19 11:25
     */
    public Boolean validateToken(String token, UserDetails userDetails) {
        String username = getUsernameFromToken(token);
        // token是否过期
        boolean tokenExpired = isTokenExpired(token);
        // 令牌是否有效
        return (username != null && username.equals(userDetails.getUsername()) && !tokenExpired);
    }


}
